# coding:utf-8

import os
import json

import set_env_path  # 请勿删除

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

from common.utils import redis_client
from common.constant import WATERING_USER_IDS

Base = declarative_base()
engine = create_engine(
    os.getenv('DATABASE_URL')
    or 'mysql+pymysql://root:@127.0.0.1/wph_www_prod?charset=utf8'
)
Session = sessionmaker(bind=engine)
session = Session()


def update_watering_user_nickname():
    """
    create by ld_wu 2017-10-30
    更新浇水用户昵称
    """
    nicknames = ["徐廷海", "胡总", "何经理", "邵", "洪总", "启光科技", "周", "鲍丹", "余总", "方生",
                 "程生", "方总", "方总", "汪先生", "姜翔", "刘道锋", "余周", "王刚", "廖晓露", "张光虹",
                 "方勇民", "徐威", "章翔", "杨志凯", "管凯", "唐俏梅", "华鑫科技", "朱江", "余芳",
                 "朱瑜", "余珍珍", "方玄亮", "姜", "余杰", "汪敬涛", "余勇", "胡经理", "叶巧礼", "王琳",
                 "虹达光电", "方路云", "余康", "余", "叶女士", "宋工", "方总", "程钰麒", "邵婧",
                 "彩程科技", "王生", "胡总", "何经理", "徐经理", "丁总", "陈越阳", "朱正", "洪生", "吕",
                 "方厚勇", "章建传", "姜女士", "方经理", "吴晓", "鹏腾科技", "洪周", "王先生", "汪景辉",
                 "周", "何总", "余萃英", "王旭敏", "余祝功", "吴燕飞", "程工", "余有华", "王平",
                 "郑瑜珍", "凌文凯", "占候斌", "卢淑娟", "方亚尧", "余晓敏", "张艳婷", "解建", "方翔",
                 "姜工", "袁总", "王瑞斌", "童经理", "胡工", "余慧成", "徐云", "徐晓瑜", "王雁",
                 "王女士", "汪晖", "王经理", "丰素文", "徐先生", "胡根华", "丰", "钱宝萍", "达宇科技",
                 "潘生", "黄丽", "周总", "唐旭东", "隆琦光电", "徐英姿", "吴之航", "张宽", "邵工",
                 "晟彩科技", "余稷", "方慧萍", "徐毅祥", "孙皓", "程总", "汪晓英", "潘", "徐生",
                 "吴鸿江", "张科峰", "章淑敏", "余慧倩", "郑哲梯", "翁飞剑", "王文富", "徐尧", "吴先生",
                 "余刚", "余清", "华彩光电", "倪哲", "管鹏穷", "余键平", "方工", "俞杨进", "方振华",
                 "汪经理", "汪建勇", "方源", "徐象飞", "汪先军", "徐总", "王旭", "夏生", "余女士",
                 "金铭光电", "陆飞腾", "余沛文", "王腾", "方青华", "徐丽玲", "陆先生", "宋经理", "方浩",
                 "方亿菲", "陈圆", "钱浩晟", "郑经理", "唐陈胜", "林沈喆", "占", "王小兵", "黄沁心",
                 "王玉林", "胡栋", "储清林", "王工", "项经理", "方敏", "钱江丽", "余晓祥", "占东仁",
                 "姜峰", "姜钦", "明诚光电", "王京华", "曹玮", "余红贵", "唐工", "王正", "方倩",
                 "王庆平", "郑征", "吴宝玉", "胡勇军", "吴宝祥", "余建", "王理", "方波", "启曦科技",
                 "管生", "徐丽", "方水盛", "方劲", "朱经理", "黄生", "林晨", "唐女士", "王总", "鲁鸿亮",
                 "王先生", "王翔", "余倩", "余荣兰", "方雨婷", "周先生", "余先生", "童源", "唐", "余晨",
                 "肖晶", "卢丹波", "章博", "陈麒年", "王国利"]
    user_ids = json.loads(redis_client.get(WATERING_USER_IDS))

    update_paid_sql = """
      update user set nickname=:nickname where id=:user_id
    """

    for idx, each in enumerate(user_ids):
        print(each)
        session.execute(
            update_paid_sql,
            {
                'nickname': nicknames[idx],
                'user_id': each
            }
        )
    session.commit()


if __name__ == '__main__':
    update_watering_user_nickname()
